package com.framework.loippi.utils;

import java.io.*;
import java.util.Enumeration;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;

/**
 * @Description: Java压缩和解压文件工具类ZipUtil
 * @author: wangcj
 * @date: 2017年6月20日 上午11:34:43
 * @version V1.0
 */
public class ZipUtil {

	private ZipUtil() {
	}

	/**
	 * 对要下载的目录进行压缩
	 * 
	 * @param downloadDir
	 *            要压缩的目录，以web工程目录为根路径
	 * @param zipFileName
	 *            要生成的目标zip文件名,后缀名必须是.zip
	 * @return
	 */
	public static String packageDir(String downloadDir, String zipFileName) {

		String realPath = null; // 最终的输出文件名
		File basePath = null; // 代表输出文件的根目录

		// 以输出的zip文件操作系统路径构造文件输出流
		BufferedOutputStream bos = null;
		try {
			realPath = downloadDir;
			if (zipFileName == null) {
				// 如果未指定文件名，则以目录名作为输出文件名
				bos = new BufferedOutputStream(new FileOutputStream(realPath + ".zip"));
			} else {
				bos = new BufferedOutputStream(new FileOutputStream(realPath + File.separator + zipFileName));
			}

			basePath = new File(realPath);

			try {
				if (new File(realPath + File.separator + zipFileName).exists()) {
					Runtime.getRuntime().exec("cmd /c delete" + realPath + File.separator + zipFileName);
					// System.out.println("del" + realPath + File.separator +
					// zipFileName);
				}
			} catch (Exception e) {
				// e.printStackTrace();
			}
		} catch (IOException e) {
			return null;
		}
		// 构造输出zip文件的ZipOutputStream对象
		ZipOutputStream zo = new ZipOutputStream(bos);

		/**
		 * 执行文件压缩，对子目录实行递归压缩，并要求输出空目录
		 */
		if (!zip(realPath, basePath, zo, zipFileName, true, false)) {
			return null;
		}

		try {
			// zo.closeEntry();
			if (null != zo)
				zo.close();
			if (null != bos)
				bos.close();
		} catch (IOException e) {
			// e.printStackTrace();
		}

		return realPath + File.separator + zipFileName;
	}

	/**
	 * 执行目录压缩的方法
	 * 
	 * @param path
	 *            要压缩的目录或文件
	 * @param basePath
	 *            如果path为目录，则该参数为new File(path)对象实例，
	 *            作用是使输出的zip文件以此目录作为根目录；如果path为 文件，则该参数为null
	 * @param zipFileName
	 *            指定的压缩文件名
	 * @param isRecursive
	 *            是否递归压缩子目录，如果path为文件，给参数应为false
	 * @param isOutBlankDir
	 *            是否输出空目录。要输出空目录则basePath不为空
	 * @return isSucessful 是否压缩成功
	 */
	private static boolean zip(String path, File basePath, ZipOutputStream zo, String zipFileName, boolean isRecursive,
			boolean isOutBlankDir) {

		boolean isSucessful = true; // 压缩操作是否成功

		File inFile = new File(path); // 创建要压缩的文件对象
		File[] files = new File[0]; // 创建保存目录及子目录内容的File数组

		/**
		 * 如果inFile是目录，则返回目录内所有的内容 否则仅保存当前文件在文件数组内
		 */
		if (inFile.isDirectory()) {
			files = inFile.listFiles();
		} else if (inFile.isFile()) {
			files = new File[1];
			files[0] = inFile;
		}

		byte[] buffer = new byte[2048]; // 创建输出文件缓冲区
		int len;

		/* 压缩文件或目录 */
		for (File file : files) {
			String pathName = null; // 被压缩文件的文件名
			if (basePath != null) {
				if (basePath.isDirectory()) { // 如果压缩的是目录
					// 截取目录内当前被遍历文件的文件名
					pathName = file.getPath().substring(basePath.getPath().length() + 1);
				} else { // 如果压缩的是文件
					// 截取文件名
					pathName = file.getPath().substring(basePath.getParent().length() + 1);
				}
			} else {
				// 获得文件名
				pathName = file.getName();
			}

			/* 如果是目录，则处理空目录并递归处理子目录，否则直接对文件进行打包压缩 */
			if (file.isDirectory()) {
				/* 压缩空目录 */
				if (isOutBlankDir && basePath != null) {
					try {
						zo.putNextEntry(new ZipEntry(pathName + File.separator));
					} catch (IOException e) {
						// e.printStackTrace();
						return false;
					}
				}
				/* 递归处理子目录 */
				if (isRecursive) {
					zip(file.getPath(), basePath, zo, zipFileName, isRecursive, isOutBlankDir);
				}
			} else {
				FileInputStream fin;
				try {
					// 创建指向文件的输入流
					fin = new FileInputStream(file);
					// 添加压缩条目
					if (!pathName.equals(zipFileName)) {
						zo.putNextEntry(new ZipEntry(pathName));
					}
					/* 向压缩条目内写入文件内容 */
					while ((len = fin.read(buffer)) > 0) {
						zo.write(buffer, 0, len);
					}
					zo.closeEntry();
					fin.close();
				} catch (Exception e) {
					// e.printStackTrace();
					return false;
				}
			}
		}
		return isSucessful;
	}

	/**
	 * 解压文件 ZipUtil.unzip("F:\\Win7壁纸.zip");
	 * 
	 * @param filePath
	 *            压缩文件路径
	 */
	public static void unzip(String filePath) {
		if (filePath.substring(0, 6).equals("file:/")) {
			filePath = filePath.substring(6);
		}
		File source = new File(filePath);
		if (source.exists()) {
			ZipInputStream zis = null;
			BufferedOutputStream bos = null;
			try {
				zis = new ZipInputStream(new FileInputStream(source));
				ZipEntry entry = null;
				while ((entry = zis.getNextEntry()) != null && !entry.isDirectory()) {
					File target = new File(source.getParent(), entry.getName());
					if (!target.getParentFile().exists()) {
						// 创建文件父目录
						target.getParentFile().mkdirs();
					}
					// 写入文件
					bos = new BufferedOutputStream(new FileOutputStream(target));
					int read = 0;
					byte[] buffer = new byte[1024 * 10];
					while ((read = zis.read(buffer, 0, buffer.length)) != -1) {
						bos.write(buffer, 0, read);
					}
					bos.flush();
				}
				zis.closeEntry();
			} catch (IOException e) {
				// throw new RuntimeException(e);
			} finally {
				closeQuietly(zis, bos);
			}
		}
	}

	/**
	 * 关闭一个或多个流对象
	 * 
	 * @param closeables
	 *            可关闭的流对象列表
	 * @throws IOException
	 */
	public static void close(Closeable... closeables) throws IOException {
		if (closeables != null) {
			for (Closeable closeable : closeables) {
				if (closeable != null) {
					closeable.close();
				}
			}
		}
	}

	/**
	 * 关闭一个或多个流对象
	 * 
	 * @param closeables
	 *            可关闭的流对象列表
	 */
	public static void closeQuietly(Closeable... closeables) {
		try {
			close(closeables);
		} catch (IOException e) {
			// do nothing
		}
	}

	/*public static void main(String[] args) {
		// packageDir("F:/Java/tomcat/tomcat7/apache-tomcat-7.0.57/webapps/admin/upload/install/","测试.zip");
		File zipFile = new File("D:/java/daogouji.zip");
		try {
			unZipFiles(zipFile, "D:/java/");
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		// System.out.println(1555555);
	}*/

	/**
	 * 解压文件 ZipUtil.unzip("F:\\Win7壁纸.zip");
	 * 
	 * @param zipFile
	 *            压缩文件路径
	 * @param descDir
	 *            解压到哪
	 */
	public static void unZipFiles(File zipFile, String descDir) throws IOException {
		File pathFile = new File(descDir);
		if (!pathFile.exists()) {
			pathFile.mkdirs();
		}
		ZipFile zip = new ZipFile(zipFile);
		for (Enumeration entries = zip.entries(); entries.hasMoreElements();) {
			ZipEntry entry = (ZipEntry) entries.nextElement();
			String zipEntryName = entry.getName();
			InputStream in = zip.getInputStream(entry);
			String outPath = (descDir + zipEntryName).replaceAll("\\*", "/");
			// 判断路径是否存在,不存在则创建文件路径
			File file = new File(outPath.substring(0, outPath.lastIndexOf('/')));
			if (!file.exists()) {
				file.mkdirs();
			}
			// 判断文件全路径是否为文件夹,如果是上面已经上传,不需要解压
			if (new File(outPath).isDirectory()) {
				continue;
			}
			OutputStream out = new FileOutputStream(outPath);
			byte[] buf1 = new byte[1024];
			int len;
			while ((len = in.read(buf1)) > 0) {
				out.write(buf1, 0, len);
			}
			in.close();
			out.close();
		}
	}
}
